import torch
model = torch.nn.Linear(10,1)
device = torch.device(0)
model=model.to(device)
print(model)
print(device)
